package echo.netty;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import io.netty.handler.codec.string.StringEncoder;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * @date： 2025/4/8 17:41
 * @description： 定义NettyEchoServer
 * @modifiedBy：
 * @version: 1.0
 */
public class NettyEchoServer {

    // 服务器监听的端口
    private static final int PORT = 8080;
    // Boss 线程组的线程数据，用于接受客户端连接
    private static final int BOSS_THREADS = 1;
    // Worker 线程组的线程数量，用于处理网络读写
    private static final int WORKER_THREADS = 2;
    // 用于统计 Worker 线程处理次数的原子整数
    private static final AtomicInteger workerThreadCounter = new AtomicInteger(0);

    public static void main(String[] args) {
        // 创建 Boss 线程组，使用 NIO 模型，指定线程数量为 BOSS_THREADS
        EventLoopGroup bossGroup = new NioEventLoopGroup(BOSS_THREADS);
        // 创建 Worker 线程组，使用 NIO 模型，指定线程数量为 WORKER_THREADS
        EventLoopGroup workerGroup = new NioEventLoopGroup(WORKER_THREADS);
        try {
            // 创建服务器启动引导类，用于配置和启动服务器
            ServerBootstrap b = new ServerBootstrap();
            // 设置使用的线程组，Boss线程组用于接收连接，Worker线程组用于处理连接后的读写操作
            b.group(bossGroup, workerGroup)
                    // 指定使用 NIO 服务器套接字通道
                    .channel(NioServerSocketChannel.class)
                    // 为每个新连接的通道添加处理器
                    .childHandler(new ChannelInitializer<SocketChannel>() {
                        @Override
                        protected void initChannel(SocketChannel ch) throws Exception {
                            // 获取通道的管道，管道是处理器链，用于处理入站和出站数据
                            ChannelPipeline pipeline = ch.pipeline();
                            // 向管道中添加字符串解码器，将接收到的字节数据解码为字符串
                            pipeline.addLast(new StringDecoder());
                            // 向管道中添加字符串编码器，将发送的字符串数据编码为字节数据
                            pipeline.addLast(new StringEncoder());
                            // 向管道中添加自定义的 Echo 处理器，用于处理接收到消息
                            pipeline.addLast(new EchoServerHandler(workerThreadCounter));
                        }
                    })
                    // 设置服务器套接字的选项，SO_BACKLOG 表示等待连接的队列长度
                    .option(ChannelOption.SO_BACKLOG, 128)
                    // 设置客户端套接字的选项，SO_KEEPALIVE 表示保持连接活跃
                    .childOption(ChannelOption.SO_KEEPALIVE, true);
            // 绑定指定的端口并开始接收连接，sync() 方法会阻塞直到绑定完成
            ChannelFuture f = b.bind(PORT).sync();
            // 打印服务器启动信息，显示监听的端口号
            System.out.println("Netty Echo Server started and listening on port " + PORT);
            // 等待服务器通道管关闭，sync() 方法会阻塞直到通道关闭
            f.channel().closeFuture().sync();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            // 优雅的关闭 worker 线程组，释放资源
            workerGroup.shutdownGracefully();
            // 优雅的关闭 Boss 线程组，释放资源
            bossGroup.shutdownGracefully();
        }
    }



}